package com.example.jpa.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.dom4j.*;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;
import org.springframework.util.StringUtils;
import org.xml.sax.SAXException;

import java.io.File;
import java.io.IOException;
import java.io.StringReader;
import java.io.StringWriter;
import java.net.URL;
import java.nio.file.Paths;
import java.util.List;
import java.util.Objects;

/**
 * @author 朝花不迟暮
 * @version 1.0
 * @date 2020/12/12 20:03
 */
public class JsonXmlUtils
{
    private static final String ENCODING = "UTF-8";

    /**
     * JSON对象转漂亮的xml字符串
     *
     * @param json JSON对象
     * @return 漂亮的xml字符串
     * @throws IOException
     * @throws SAXException
     */
    public static String jsonToPrettyXml(JSONObject json) throws IOException, SAXException
    {
        Document document = jsonToDocument(json);

        /* 格式化xml */
        OutputFormat format = OutputFormat.createPrettyPrint();

        // 设置缩进为4个空格
        format.setIndent(" ");
        format.setIndentSize(4);

        StringWriter formatXml = new StringWriter();
        XMLWriter writer = new XMLWriter(formatXml, format);
        writer.write(document);

        return formatXml.toString();
    }

    /**
     * JSON对象转xml字符串
     *
     * @param json JSON对象
     * @return xml字符串
     * @throws SAXException
     */
    public static String JsonToXml(JSONObject json) throws SAXException
    {
        return jsonToDocument(json).asXML();
    }

    /**
     * JSON对象转Document对象
     *
     * @param json JSON对象
     * @return Document对象
     * @throws SAXException
     */
    public static Document jsonToDocument(JSONObject json) throws SAXException
    {
        Document document = DocumentHelper.createDocument();
        document.setXMLEncoding(ENCODING);

        // root对象只能有一个
        /*for (String rootKey : json.keySet())
        {
            Element root = jsonToElement(json.getJSONObject(rootKey), rootKey);
            document.add(root);
            break;
        }*/
        Element root = jsonToElement(json, "root");
        document.add(root);
        return document;
    }

    /**
     * JSON对象转Element对象
     *
     * @param json     JSON对象
     * @param nodeName 节点名称
     * @return Element对象
     */
    public static Element jsonToElement(JSONObject json, String nodeName)
    {
        Element node = DocumentHelper.createElement(nodeName);
        for (String key : json.keySet())
        {
            Object child = json.get(key);
            if (child instanceof JSONObject)
            {
                node.add(jsonToElement(json.getJSONObject(key), key));
            } else if (child instanceof List)
            {
                for (int i = 0; i < ((List) child).size(); i++)
                {
                    Object o = ((List) child).get(i);
                    String str = JSONObject.toJSONString(o);
                    JSONObject jsonObject = JSON.parseObject(str);
                    // jsonToElement(jsonObject, "item");
                    node.add(jsonToElement(jsonObject, key));
                }
            } else
            {
                Element element = DocumentHelper.createElement(key);
                element.setText(json.getString(key));
                node.add(element);
            }
        }

        return node;
    }

    /**
     * xml转json
     *
     * @param xmlStr
     * @return
     * @throws DocumentException
     */
    public static JSONObject xml2Json(String xmlStr) throws DocumentException
    {
        Document doc = DocumentHelper.parseText(xmlStr);
        JSONObject json = new JSONObject();
        dom4j2Json(doc.getRootElement(), json);
        return json;
    }

    /**
     * xml转json
     *
     * @param element
     * @param json
     */
    public static void dom4j2Json(Element element, JSONObject json)
    {
        //如果是属性
        for (Object o : element.attributes())
        {
            Attribute attr = (Attribute) o;
            if (!StringUtils.isEmpty(attr.getValue()))
            {
                json.put("@" + attr.getName(), attr.getValue());
            }
        }
        List<Element> chdEl = element.elements();
        if (chdEl.isEmpty() && !StringUtils.isEmpty(element.getText()))
        {//如果没有子元素,只有一个值
            json.put(element.getName(), element.getText());
        }

        for (Element e : chdEl)
        {//有子元素
            if (!e.elements().isEmpty())
            {//子元素也有子元素
                JSONObject chdjson = new JSONObject();
                dom4j2Json(e, chdjson);
                Object o = json.get(e.getName());
                if (o != null)
                {
                    JSONArray jsona = null;
                    if (o instanceof JSONObject)
                    {//如果此元素已存在,则转为jsonArray
                        JSONObject jsono = (JSONObject) o;
                        json.remove(e.getName());
                        jsona = new JSONArray();
                        jsona.add(jsono);
                        jsona.add(chdjson);
                    }
                    if (o instanceof JSONArray)
                    {
                        jsona = (JSONArray) o;
                        jsona.add(chdjson);
                    }
                    json.put(e.getName(), jsona);
                } else
                {
                    // 当反向解析的时候如果数组标签下只有一条数据, 他会转成jsonObject, 这是不对的
                    // if(!chdjson.isEmpty()){
                    //     json.put(e.getName(), chdjson);
                    // }
                    if (!chdjson.isEmpty())
                    {
                        if (!Objects.isNull(chdjson.get("children")))
                        {
                            JSONArray jsonArray = new JSONArray();
                            jsonArray.add(chdjson);
                            json.put(e.getName(), jsonArray);
                        } else
                        {
                            json.put(e.getName(), chdjson);
                        }
                    }

                }
            } else
            {
                //子元素没有子元素
                for (Object o : element.attributes())
                {
                    Attribute attr = (Attribute) o;
                    if (!StringUtils.isEmpty(attr.getValue()))
                    {
                        json.put("@" + attr.getName(), attr.getValue());
                    }
                }
                if (!e.getText().isEmpty())
                {
                    json.put(e.getName(), e.getText());
                }
            }
        }
    }

    /**
     * XML字符串转JSON对象
     *
     * @param xml xml字符串
     * @return JSON对象
     * @throws DocumentException
     */
    public static JSONObject xmlToJson(String xml) throws DocumentException
    {
        JSONObject json = new JSONObject();

        SAXReader reader = new SAXReader();
        Document document = reader.read(new StringReader(xml));
        Element root = document.getRootElement();

        json.put(root.getName(), elementToJson(root));

        return json;
    }

    /**
     * Element对象转JSON对象
     *
     * @param element Element对象
     * @return JSON对象
     */
    public static JSONObject elementToJson(Element element)
    {
        JSONObject json = new JSONObject();
        for (Object child : element.elements())
        {
            Element e = (Element) child;
            if (e.elements().isEmpty())
            {
                json.put(e.getName(), e.getText());
            } else
            {
                json.put(e.getName(), elementToJson(e));
            }
        }

        return json;
    }

    /**
     * 文件内容转换成字符串
     *
     * @param filePath 文件路径
     * @return 内容字符串
     * @throws IOException
     */
    public static String fileToString(URL filePath) throws IOException
    {
        return IOUtils.toString(filePath, ENCODING);
    }

    /**
     * 文件内容转换成字符串
     *
     * @param filePath 文件路径
     * @return 内容字符串
     * @throws IOException
     */
    public static String fileToString(String filePath) throws IOException
    {
        return IOUtils.toString(Paths.get(filePath).toUri(), ENCODING);
    }

    /**
     * 字符串输出到文件
     *
     * @param str      字符串内容
     * @param filePath 文件路径
     * @throws IOException
     */
    public static void stringToFile(String str, String filePath) throws IOException
    {
        FileUtils.writeStringToFile(Paths.get(filePath).toFile(), str, ENCODING);
    }

    /**
     * 字符串输出到文件
     *
     * @param str      字符串内容
     * @param filePath 文件路径
     * @throws IOException
     */
    public static void stringToFile(String str, URL filePath) throws IOException
    {
        FileUtils.writeStringToFile(new File(filePath.getPath()), str, ENCODING);
    }

    /**
     * 字符串输出到文件
     *
     * @param str  字符串内容
     * @param file 输出文件
     * @throws IOException
     */
    public static void stringToFile(String str, File file) throws IOException
    {
        FileUtils.writeStringToFile(file, str, ENCODING);
    }

    public static void main(String[] args) throws IOException, SAXException, DocumentException
    {
        String json="{\"message\": \"用户信息获取成功\",\"code\": \"1000\",\"children\": [{\"user_id\": \"454645646456\",\"username\": \"月忆秋思\",\"xfbUserType\":0,\"realname\":\"沈万三\",\"telephone\":\"99944454564\",\"tel400\":\"4765132345\",\"sex\":1,\"city\":\"city\",\"projname\":\"海上传奇\",\"newcode\":\"2313145646546\"}]}";
        String xml = jsonToPrettyXml(JSONObject.parseObject(json));
        System.out.println("xml = " + xml);

        JSONObject jsonObject = xml2Json(xml);
        System.out.println("jsonObject = " + jsonObject);
    }
}
